from app.schema.base import BaseSchema
class MilvusConfig(BaseSchema):
    host:str = '192.168.2.45'
    port:str  = '19530'
    user:str  = 'root'
    password:str = "Milvus"

class MilvusIndexParam(BaseSchema):
    index_type:str = 'FLAT'
    params:dict = {
        "nlist": 1024
    }
    metric_type:str='COSINE'

class MilvusSearchParam(BaseSchema):
    metric_type:str='COSINE'
    expr:str=None
    # params:dict={
    #     "nprobe": 12,
    #     "top_k":5
    # }